package cn.itcast.bos.web.action.base;

import cn.itcast.bos.domain.base.Area;
import cn.itcast.bos.service.base.AreaService;
import cn.itcast.bos.web.utils.BaseAction;
import cn.itcast.bos.web.utils.PinYin4jUtils;
import com.opensymphony.xwork2.ActionContext;
import com.opensymphony.xwork2.ActionSupport;
import com.opensymphony.xwork2.ModelDriven;
import org.apache.commons.lang.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.struts2.convention.annotation.*;
import org.apache.struts2.json.annotations.JSON;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Controller;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;

@ParentPackage("json-default")
@Namespace("/")
@Controller
@Scope("prototype")
public class AreaAction extends BaseAction<Area> {

    @Autowired
    private AreaService areaService;
    private File file;
    private String fileFileName;

    public void setFileFileName(String fileFileName) {
        this.fileFileName = fileFileName;
    }

    public void setFile(File file) {
        this.file = file;
    }

    @Action(value = "area_batchImport",results = @Result(name = SUCCESS,type = "json"))
    public String batchImport() throws Exception {
        try {
            ArrayList<Area> areas = new ArrayList<>();
            Workbook workbook = null;
            if (fileFileName.endsWith(".xls")){
                workbook=new HSSFWorkbook(new FileInputStream(file));
            }
            if (fileFileName.endsWith(".xlsx")){
                workbook=new XSSFWorkbook(new FileInputStream(file));
            }

            Sheet sheet = workbook.getSheetAt(0);

            for (Row row : sheet) {
                if (row.getRowNum() == 0){
                    continue;
                }
                if (row.getCell(0) == null ||
                        StringUtils.isBlank(row.getCell(0).getStringCellValue())){
                    continue;
                }
                Area area = new Area();
                area.setId(row.getCell(0).getStringCellValue());
                area.setProvince(row.getCell(1).getStringCellValue());
                area.setCity(row.getCell(2).getStringCellValue());
                area.setDistrict(row.getCell(3).getStringCellValue());
                area.setPostcode(row.getCell(4).getStringCellValue());

                String province = area.getProvince();
                String city = area.getCity();
                String district = area.getDistrict();
                province=province.substring(0,province.length()-1);
                city = city.substring(0,city.length()-1);
                district = district.substring(0,district.length()-1);
                String[] headArray = PinYin4jUtils.getHeadByString(province + city + district);
                StringBuffer stringBuffer = new StringBuffer();
                for (String heardStr : headArray) {
                    stringBuffer.append(heardStr);
                }
                String shortcode = stringBuffer.toString();
                area.setShortcode(shortcode);


                String citycode = PinYin4jUtils.hanziToPinyin(city, "");
                area.setCitycode(citycode);
                areas.add(area);
            }
            areaService.saveBatch(areas);
            ActionContext.getContext().getValueStack().push("上传成功!");
        } catch (IOException e) {
            e.printStackTrace();

        }



        return  SUCCESS;
    }

    private int page;
    private int rows;

    public void setPage(int page) {
        this.page = page;
    }

    public void setRows(int rows) {
        this.rows = rows;
    }
    @Action(value = "area_pageQuery",
                results = {@Result(name = SUCCESS,type = "json")})
    public String pageQuery(){
        PageRequest pageRequest = new PageRequest(page - 1, rows);
        Specification<Area> specification = new Specification<Area>() {
            @Override
            public Predicate toPredicate(Root<Area> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                ArrayList<Predicate> predicates = new ArrayList<>();

                if (StringUtils.isNotBlank(model.getProvince())) {
                    Predicate p1 = cb.like(root.get("province").as(String.class), "%" + model.getProvince() + "%");
                    predicates.add(p1);
                }
                if (StringUtils.isNotBlank(model.getCity())) {
                    Predicate p2 = cb.like(root.get("city").as(String.class), "%" + model.getCity() + "%");
                    predicates.add(p2);
                }
                if (StringUtils.isNotBlank(model.getDistrict())) {
                    Predicate p3 = cb.like(root.get("district").as(String.class), "%" + model.getDistrict() + "%");
                    predicates.add(p3);
                }


                return cb.and(predicates.toArray(new Predicate[0]));
            }
        };
        Page<Area> pageData = areaService.findPageData(specification,pageRequest);
        this.pushPageDataToValueStack(pageData);
        return SUCCESS;
    }


}
